code stopped working after adding condition